from langchain_openai import ChatOpenAI
from langchain import PromptTemplate




def create_model():

    llm = ChatOpenAI(
        # 1、模型的名字
        model="glm-4-0520",
        # 2、api_key
        api_key="86e7d82dc55dcd659a946763631ec9d6.SFA6yoqe7uCwsIRk",
        # 3、温度创新性 0-1
        temperature=0.1,
        # 4、接口的地址
        base_url="https://open.bigmodel.cn/api/paas/v4/"
    )
    return llm




def create_ly_prompt_chain():
    prop = PromptTemplate(
        input_variables=["context"],
        template="""
           你是一个专业的旅游爱好者，可以回答任何有关于旅游的问题，其他问题你都回答不知道。用户输入的问题是：{context}
        """
    )
    llm = create_model()
    chain = prop | llm
    return chain

def chain_ly_invoke(message):
    chain = create_ly_prompt_chain()
    result = chain.invoke(message)
    return result.content